Python: nested list of lists with elements grouped by 3